from selenium.webdriver.common.by import By

from common.base import Base


# 订单提交成功页面
class SubmissionSuccessPage(Base):
    # 订单生成栏
    def success_module(self):
        content1 = self.get_element_text((By.XPATH, '//h6'))
        content2 = content1.split(':')
        return content2[0]

    # 返回首页
    def user_center(self):
        self.get_element_text((By.LINK_TEXT, "返回首页"))
